# 数据加载和检查点处理
import pprint
import torch

# 加载训练好的模型检查点
checkpoint = torch.load('SEED_jointly_checkpoint/checkpoint_1.pkl')
print("Checkpoint keys:", checkpoint.keys())
print("\nModel state dict keys:")
pprint.pprint(list(checkpoint['model'].keys()))
print(f"\nBest Accuracy: {checkpoint['ACC']:.4f} at Epoch {checkpoint['epoch']}")